# coding=utf-8 
'''
 @Author  : 测试工程师成长之路
 @Time    : 2021/7/2
 @Desc    : Airtest启动器--添加自定义的命令行参数
 @Software: PyCharm
'''

from airtest.cli.runner import AirtestCase, run_script
from airtest.cli.parser import runner_parser


class CustomAirtestCase(AirtestCase):
    def setUp(self):
        # 在air脚本运行之前获取这个自定义的参数
        if self.args.env:
            self.scope['env']=self.args.env


if __name__ == '__main__':
    ap = runner_parser()
    # 添加自定义的命令行参数
    ap.add_argument(
        "--env", help="environment configuration parameters, such as SIT, UAT, PROD")
    args = ap.parse_args()
    run_script(args, CustomAirtestCase)
